// Copyright (c) 2024 Huawei Technologies Co., Ltd.
// openFuyao is licensed under Mulan PSL v2.
// You can use this software according to the terms and conditions of the Mulan PSL v2.
// You may obtain a copy of Mulan PSL v2 at:
//          http://license.coscl.org.cn/MulanPSL2
// THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
// EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
// MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
// See the Mulan PSL v2 for more details.

package lm

import (
	corev1 "k8s.io/api/core/v1"
)

const (
	acceleratorLabelKey = "accelerator"
)

func newAcceleratorLabeler(node *corev1.Node) Labeler {
	labels := getAccelerator(node)
	return labels
}

func getAccelerator(node *corev1.Node) Labeler {
	for key := range node.Labels {
		if value, ok := npuDeviceTypes[key]; ok {
			return Labels{acceleratorLabelKey: value}
		}
	}

	return Labels{acceleratorLabelKey: "unknown"}
}
